package licode.unisop.platform.infrastructure.util;

import com.baomidou.mybatisplus.core.metadata.IPage;
import licode.unisop.unilite.dp.PageData;

import java.util.List;
import java.util.function.Function;

/**
 * Mybatis服务工具类
 *
 * @author WR
 * @since 2016/7/12 8:02
 */
public class MybatisPageUtil {
    public static <K, T> PageData<K> convertPage(IPage<T> page, Function<List<T>, List<K>> call) {
        PageData<K> pageData = new PageData<>();
        if (page != null) {
            pageData.setCurrent(page.getCurrent());
            pageData.setSize(page.getSize());
            pageData.setTotal(page.getTotal());
            pageData.setPages(page.getPages());
            pageData.setData(call.apply(page.getRecords()));
        }
        return pageData;
    }
}
